# 导入操作相关的功能模块
import os
from cozepy import Coze, TokenAuth, COZE_CN_BASE_URL

# 导入环境变量
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()


# 获取工作空间的列表
def get_space_list():

    # 通过环境变量引入密钥，访问 coze.cn 服务
    # 声明访问令牌
    api_token = os.getenv("COZE_API_TOKEN")

    # 如果令牌为空，提示用户
    if not api_token:
        print("请先设置一下个人令牌")

    # 初始化coze客户端
    coze = Coze(
        # 声明令牌
        auth = TokenAuth(api_token),
        # 声明域名
        base_url = COZE_CN_BASE_URL
    )

    # 调用coze客户端获取空间列表
    try:
        space_list = coze.workspaces.list()

        if hasattr(space_list, "items"):
            spaces = space_list.items
            print(spaces)

    except Exception as e:
        print(f"coze_sdk: 获取空间列表失败, 错误为: {str(e)}")

if __name__ == "__main__":
    get_space_list()